package cn.edu.xjtu.hash.hashTable.findRestaurant;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 两个列表的最小索引总和
 * 
 * 假设Andy和Doris想在晚餐时选择一家餐厅，并且他们都有一个表示最喜爱餐厅的列表，每个餐厅的名字用字符串表示。
 * 你需要帮助他们用最少的索引和找出他们共同喜爱的餐厅。 如果答案不止一个，则输出所有答案并且不考虑顺序。 你可以假设总是存在一个答案。
 */
public class Solution {
  public static void main(String[] args) {
    String[] list1 = new String[] { "Shogun", "Tapioca Express", "Burger King", "KFC" };
    String[] list2 = new String[] { "Piatti", "The Grill at Torrey Pines", "Hungry Hunter Steakhouse", "Shogun" };
    System.out.println(Arrays.toString(findRestaurant(list1, list2)));
  }

  /**
   * 
   * @param list1 Andy喜欢的餐厅
   * @param list2 Doris喜欢的餐厅
   * @return 两个人共同喜欢且下标之和较小的餐厅列表
   */
  public static String[] findRestaurant(String[] list1, String[] list2) {
    List<String> ret = new ArrayList<>();
    Map<String, Integer> map = new HashMap<>();
    // 将andy喜欢的餐厅和他的下标存入哈希表中
    for (int i = 0; i < list1.length; i++) {
      map.put(list1[i], i);
    }
    // 遍历Doris喜欢的餐厅,得到两人喜欢餐厅的最小下标和
    int minIndex = 2000;
    for (int i = 0; i < list2.length; i++) {
      if (map.containsKey(list2[i])) {
        if (map.get(list2[i]) + i < minIndex) {
          minIndex = map.get(list2[i]) + i;
        }
      }
    }
    for (int i = 0; i < list2.length; i++) {
      if (map.containsKey(list2[i])) {
        if (map.get(list2[i]) + i == minIndex) {
          ret.add(list2[i]);
        }
      }
    }
    return ret.toArray(new String[] {});
  }
}
